from pydantic import BaseModel


class HealthData(BaseModel):
    gender: str
    age: int
    hypertension: int
    heart_disease: int
    ever_married: str
    work_type: str
    Residence_type: str
    avg_glucose_level: float
    bmi: float
    smoking_status: str
    mode: str


class SalaryData(BaseModel):
    sex: str
    age: int
    workclass: str
    fnlwgt: int
    education: str
    education_num: int
    marital_status: str
    occupation: str
    relationship: str
    race: str
    capital_gain: int
    capital_loss: int
    hours_per_week: int
    native_country: str
    mode: str
